Updated SIR Model