Added spread of infection to the simulation.