work on assignment 2