Update for Assignment 1