Modeling spread of COVID-19