Added code to enable moving reovered individuals back to susceptible compartment. The "removed" compartment is broken and doesn't work (but the calculations are correct).