add tests for code coverage