Working on assignment 7