Updates for Assignment 2