Added code coverage test