Simulating infection spread using a particle simulation model created with Matplotlib Animations. Still working on colors of particles changing in saved videos; works when running model with animation=False