AI - improved upgrade selection.