Add matrix_square (i.e. A * trans(A))