Fixed the AI, is now smart