Необходимо переделать код. Квадратная матрица

 static void diagon(int n, int[,] matr, int[] diag)
        {
            for (int i = 0; i < n; i++)
            {
                int s = 0;
                for (int j = 0; j < i + 1; j++)
                {
                    int i1 = n - 1 - i + j;
 
                    s += matr[i1, j];
                }
                diag[i] = s;
            }
            for (int i = n; i < 2 * n - 1; i++)
            {
                int s = 0;
                int i1 = i - n;
                for (int j = 0; j < n - i1 - 1; j++)
                {
                    int i2 = j + i1 + 1;
                    s += matr[j, i2];
                }
                diag[i] = s;
            }
        }
 
             
        static void Problem3()
        {
            int N = 4;
            int[,] matrix = new int[N, N];
            int[] diag = new int[2 * N - 1];
            Random rnd = new Random();
            Console.WriteLine("Матрица");
            for (int i = 0; i < N; i++)
            {
                for (int t = 0; t < N; t++)
                {
                    matrix[i, t] = rnd.Next(0, 10);
                    Console.Write(matrix[i, t] + "\t");
                }
                Console.WriteLine();
            }
            Console.WriteLine();
            diagon(N, matrix, diag);
            for (int i = 0; i < 2 * N - 1; i++)
                Console.WriteLine("Номер диагонали = {0}, Сумма элементов = {1}", i + 1, diag[i]);
            Console.ReadKey();
        }

using System;
using System.Collections.Generic;
 
namespace ConsoleApp1
{
    class Program
    {
        static void Main(string[] args)
        {
            int[,] mx =
            {
                {1,   2,  3,  4,  5},
                {6,   7,  8,  9, 10},
                {11, 12, 13, 14, 15},
                {16, 17, 18, 19, 20},
                {21, 22, 23, 24, 25}
            };
            Console.WriteLine(String.Join(" ", CollateralSum(mx)));
            Console.WriteLine("Контрольные значения: \n1 8 21 40 65 64 57 44 25");
            Console.ReadKey();
 
        }
 
        static int[] CollateralSum(int[,] mx)
        {
            if (mx.GetLength(0) != mx.GetLength(1))
                throw new ArgumentException("Длины обоих измерений должны быть одинаковыми", "mx");
            var result = new List<int>();
            for (int i = 0; i < mx.GetLength(0) + mx.GetLength(1) - 1; i++)
            {
                int startCol = i < mx.GetLength(0) ? 0 : i - mx.GetLength(0) + 1;
                int startRow = startCol == 0 ? i : mx.GetLength(1) - 1;
                int curCol = startCol, curRow = startRow, sum = 0;
                do
                {
                    sum += mx[curRow--, curCol++];
                } while (curRow >= startCol);
                result.Add(sum);
            }
            return result.ToArray();
        }
    }
}