improved the AI a bit