Increase test coverage
Increase branch coverage to 100%