add tests

add tests and check coverage