Working version but not finished exercise4