More code coverage added