first try to improve AI