Нужен код который возвращает максимальное из чисел, находящихся на главной диагонали квадратного двумерного массива

@Test
    public void testDiagonalMaxMatrix() {
        FirstSteps firstSteps = new FirstSteps();
        int[][] matrix1 = {{1, 2, 3}, {3, 4, 5}, {6, 7, 8}};
        assertEquals(8, firstSteps.diagonalMax(matrix1));
        int[][] matrix2 = {{100, 2, 3}, {3, 4, 5}, {6, 7, 800}};
        assertEquals(800, firstSteps.diagonalMax(matrix2));
        int[][] matrix3 = {{}};
        assertEquals(Integer.MIN_VALUE, firstSteps.diagonalMax(matrix3));
        int[][] matrix4 = {{-1, -2, 3}, {3, -4, 5}, {6, -7, -8}};
        assertEquals(-1, firstSteps.diagonalMax(matrix4));
    }

public int max(int[][] matrix) {
        int max = -1;
        if (matrix.length == 1 && matrix[0].length == 0) return Integer.MIN_VALUE;;
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                if (matrix[i][j] > max) {
                    max = matrix[i][j];
                }
            }
        }
        System.out.println("max value in myArray: " + max);
        return max;
    }

for (int j = 0; j < matrix.length; j++) {
     if (matrix[j][j] > max) max = matrix[j][j];
}

public int max(int[][] matrix) {
        int max = -1;
        if(matrix.length == 0 || matrix.length == 1) return Integer.MIN_VALUE;
        for (int i = 0; i < matrix.length; i++) {
                if (matrix[i][i] > max) {
                    max = matrix[i][i];
                }
        }
        return max;
    }