Changes made to complete Programming Assignment 2