Update ai.cpp