Adding tests to Chapter_02