mistakes in hw 4
scratch nb for hw 4 stochastic