Improved stochastic