Writting the test classes