solution to Programming Assignment 2