Unit AI updated