Update assign3_3.py