add test code

test code