Added the test classes