Update Assignment4