Add the Test class