Finished exercise 2 from chapter 1