Update to CaseStudy.py