Small update to AI