Minor changes to AI system