Update main_script.py