Update assign3_1.py