Update lab1_part3.py