SIR Model added