Finalize the diagonal function
Add "columns" and "diagonals"
Complete the solution