fix how epidemics are handled
fix bug in how second infection was handled