Chapter 2 : Iteration 2 : My changes