Update FinalProject_Prob1.py