add the function of AI