Finished hw4 part 1