Update Assignment_2.py