add unit tests for 100% code coverage