Update Assignment3.py