Update Assignment 2