add unit test to code
add unit test to code