Minor update + AI