add class for tests