add test
add code coverage