Adds tests for Chapter 2