I've recently been experimenting with the RnSIR model. See my slides here