Update Project_3_Prob2.py