Add ability to set the logger.