Started integrating the ai