Minor AI engine changes