TODO: New function for limitedInfection