Update logging functionality