Add back matrix for comparison