Greatly improve ai