Update final_project.py