well one thing that is cheap and a easy fix on the lightswitch power delivery of 2 strokes is a heavier flywheel, my dad was in the same boat as you, he loves how easy the 4 strokes are to ride, but we cannot financially support the costs of 4 strokes. He had a 450, sold it, 2 weeks later it blew up and needed 1800 in parts, he bought a yz250, and coudln't ride it, he was all over the place, blowing berms overjumping stuff, he's been on a big 4 stroke for about 8 years before the yz250 so he had to re-learn how to ride. We put the biggest flywheel that stealthy makes for it which is a 13oz and he loves the yz250 now, it has the same power of a regular yz250 its just smoother and way easier to ride, it was i think 119.99, and it takes 5 minutes to install. Tames the hit way down and makes it very tractable while still having the hit if you want it with a flick of the clutch and the crack of the throttle. Best investment he said he's ever made, and he said he will never ride a 4 stroke mx bike again because he says the yz is perfect.
Don't feel bad man, they're not the easiest bikes to ride, which is why most people like 4 strokes over 2 strokes, but a 2 strokes power delivery is easy to change and relatively cheap, they each have a very different riding style, you have to be more aggressive on a 2 stroke and you have to be on your toes the entire time or it will bite you, more braking, more accelerating, it will skate around more and won't feel as planted. The flywheel kind of slows everything down, it makes it feel more planted because you're accelerating without as much wheelspin but it still will move around a little more than a 4 stroke just because of the weight difference and having more power than a 250f
try and find somebody in your area with a ktm trail 2 stroke, like a 250exc or 300 exc, they're smooth and tractable like a 4 stroke with plenty of power, this will give you an idea of what a flywheel weight can do to your bike