Add tests to increase code coverage