Small AI improvements.