work on the assignment 2