AI improvements