Вывести на экран сумму элементов лежащих на главной диагонали

Random r = new Random();
n=10;
int[][] a = new int[n][n];
for (int i=0;i<n;i++){
for (int j=0;j<n;j++){
a[i][j]=r.nextInt(100);
System.out.print(a[i][j]+"\t");}
System.out.println();}
 
int s=0;
for (int i=0;i<n;i++)
for (int j=0;j<n;j++)
if (i=j) s+=a[i][j];
 
System.out.println("Сумма элементов на главной диагонали "+s);

import java.util.Arrays;
 
public class Task006 {
    public static void main(String[] args) {
        int n = 5;
        int[][] matrix = createMatrix(n, n, 0, 9);
        System.out.println("Дана матрица:\n" + matrixToString(matrix));
 
        int sum = sumMainDiagonal(matrix);
        System.out.println("Сумма элементов главной диагонали матрицы: " + sum);
    }
 
    public static int sumMainDiagonal(int[][] matrix) {
        int sum = 0;
        if (matrix != null) {
            for (int i = 0; i < matrix.length && i < matrix[i].length; i++) {
                sum += matrix[i][i];
            }
        }
        return sum;
    }
 
    public static int[][] createMatrix(int n, int m, int min, int max) {
        int[][] matrix = n > 0 && m > 0 ? new int[n][m] : new int[0][0];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                matrix[i][j] = (int) (Math.random() * (Math.abs(max - min) + 1)) + min;
            }
        }
        return matrix;
    }
 
    public static String matrixToString(int[][] matrix) {
        StringBuilder sb = new StringBuilder();
        if (matrix != null) {
            for (int[] array : matrix) {
                sb.append(Arrays.toString(array)).append("\n");
            }
        }
        return sb.toString();
    }
}