Update Project_1_Prob2.py