actually fixed the diag function
Merge pull request #15 from AnnaKalkanis/diagonal_bug_fix_anna

actually fixed the diag function