Add logger support