Enhance logging capabilities