add test class and method