Another improvement of the AI!