Added ai classes