Update final_exam.py