Small AI changes