Add functionality to generate individuals and to randomly select individuals to infect at time 0