An Infectious Agent Simulation in PyGame