SIR model in class example