Minor AI changes