A more salient Star Trek episode about rogue AI a year later would be The Ultimate Computer, where the M5 multitronic computer is placed in control of the Enter

A more salient Star Trek episode about rogue AI a year later would be The Ultimate Computer, where the M5 multitronic computer is placed in control of the Enterprise and performs superbly in battle simulations, leading to worries that it would obsolete Captain Kirk and the rest of the crew, until it goes out of control and starts killing people and has to be defeated by human ingenuity.