Adding Test class