Update HW1_script.py