Adding code coverage and test.