stochastic models python