updated setup.py to install stochastic module