Update hw3_1.py