Add callbacks to early exit functions