you don't need W afterwards. Make the ** 2 inplace with `np.power(W, 2, W)`