Add hooking support