Add custom logger support