Added optional argument to Diag() to access super/sub diagonals.