Add editFocusGained hook