Started working on the ai