added ai integration