work on assignment 4