add stochastic environment: attempt 2