Fixed SIR model a bit.