AI upgrade