Update assign7_2.py