Increase test coverage

Test coverage is at 100% for the first time