Add callbacks to existing methods