Added code to infect.