Update python_script.py