Add class to test file