Added some Test class