#5: Increase test coverage