Add AI logic