Add custom logging functionality (#2155)