Improved ai play