Adding test coverage
Increasing code coverage