AI for the Battleship flow