adding the assignment 2