finish work on assignment 2