Update stochastic.rb