Added WS generation, added prevention to SIR model.