Update the assignment 2