Infection Spread Simulation - Coding problem | OneNoughtOne