Edit program for ProgrammingAssignment2