If I make it run on a cell after plt.plot(ki0), output_size = gm.shape[0], I get a NameError: name 'gm' is not defined. This error says that gm does not exist, but where can I find gm?