implement unit test
add test data