Same idea, remove function, use hook.