Update AI mode with visual cue and other minor fixes