Improve AI

Well, it is worse now...