Add support for custom log output