Week 4- work on models and modifying the epidemic base model