Changes made for Programming Assignment 2