Add Test class