Working on assignment 6