Small AI tweaks