SIR Model with Python (Stochastic Version)