Assignment 1 update