Problem Set 1 Basic Covid Model