add test class

add test class