further optimize AI