Add tests for Chapter 3