Should we just use np.dot and np.diag to be clear? I don't know if we save all that much by using dot and diag