Added the diagonal of a matrix to the Trace class