working on assignment 2