Improve AI more