refactoring, add new test class method