Added Test class