I usually avoid any use of np.matrix. Could we make it an array?