Here's the class for the AI I mentioned