Working on assignment 3