This would require adding a boolean method to EntityArmyMember, then changing your AI checks to simply check for that. It would make it so each specific case can be handled within the entity's instance rather than by the AI classes, making the overall design a lot better, and much easier for addons to extend the functionality.