Add SIR calculation to SIRFunction$