Update Project_4_Prob3a.py