Update Project_4_Prob1b.py