need test class