Improved code coverage to 100%.