Re-add Azure diagnostics tracing