Slight AI update
I forget