minor fixes
working SIR model