Update Assignment2.py