Adding code for code coverage