Let's make the AI logic clearer