HW 6 - COVID-19 Simulation