Adding code to test coverage