Improved the AI.