Add instrumentation hooks