Add new functionality in the Chapter 2