add test code

add test code