Update Assignment 4