Working on assignment 1