Update Assignment1.py