more work on the stochastic example