Updating Assignment 2.