Adding work to Programming Assignment 2