Minor changes in Programing Assignment 1