changes how diagonal rows are extracted
adds method for getting just the diagonal rows