Update HW3_1.py