Add callbacks for instrumentation