Add TestCategory class