changes made to the end of the hw4 part 1 assignment