Update Assignment4.py