Update to Chapter 4 Exercise 2