AI update

small updates