working on assignment 4