Added changes according to 1st example in book