Add diagonal

Add diagonal function (without axis parameter) and tests for small 2x2 matrix, random 15x10, and larger random 16x16