add AI code