Update final_script.py