Update ngd_stochastic.py