Updated the assignment