more work on assignment 3