Working on assignment 2