Updating for assignment 2.