Added code coverage
Fixed test