Update assign5_2.py