Update main_script_week1.py

Full code