more work on assignment 2