Added classes to support the previous test case so that code compiles