Remade TraitAI and made some changes to TraitRegistry