Finished exercise 6 part 2