Update Assignment 3