Incorporating Stochasticity in Simple Models of Disease Spread