add ai logic