Finished exercise 4