Update to Chapter 4 Exercise 1