Minor Ai changes