Updates Assignment 2