Work on assignment 4