Assignment 2 update