minor changes to AI