Update Project_2_Prob3b.py