Como somar os elementos da diagonal principal de uma matriz em Ruby

# vamos declarar e construir uma matriz de três linhas e três colunas
matriz = Array.new(3){Array.new(3)}
soma_diagonal = 0; # guarda a soma dos elementos na diagonal principal

# vamos ler os valores para os elementos da matriz
for i in (0..2) # linhas
  for j in (0..2) # colunas
    printf("Valor para a linha %d e coluna %d: ", i, j) 
    matriz[i][j] =  gets.chomp.to_i
  end
end

# vamos mostrar a matriz da forma que ela
# foi informada
print("\n")
for i in (0..2) # linhas
  for j in (0..2) # colunas
    printf("%5d ", matriz[i][j])
  end
  print("\n")
end

# vamos calcular a soma dos elementos da diagonal   
# principal
for i in (0..2) # linhas
  for j in (0..2) # colunas
    if(i == j)
      soma_diagonal = soma_diagonal + matriz[i][j]
    end
  end
end

# e mostramos o resultado
printf("\nA soma dos elementos da diagonal principal é: %d",
  soma_diagonal)

Informe o valor para a linha 0 e coluna 0: 3
Informe o valor para a linha 0 e coluna 1: 7
Informe o valor para a linha 0 e coluna 2: 9
Informe o valor para a linha 1 e coluna 0: 2
Informe o valor para a linha 1 e coluna 1: 4
Informe o valor para a linha 1 e coluna 2: 1
Informe o valor para a linha 2 e coluna 0: 5
Informe o valor para a linha 2 e coluna 1: 6
Informe o valor para a linha 2 e coluna 2: 8

    3     7     9 
    2     4     1 
    5     6     8 

A soma dos elementos da diagonal principal é: 15