Adding a test for spreadInfection