add AI main code