Updated Chapter 2 Exercise 4