Exercise 4 changes