more work on assignment 2
assignment 3, task 1