Leer matriz y sumar los numeros en la diagonal

import java.util.*;
public class sumad{
  public static void main(){
      int a[][]= new int [3][3];
      int suma,i,j;
      Scanner teclado = new Scanner (System.in);
      suma=0;
      System.out.println("Escribe los valores de las casillas");
      for (i=0;i<3;i++){
          for (j=0;j<3;j++){
               System.out.println("casilla "+i+","+j);
              a[i][j]= teclado.nextInt();
              if (i==j){
                  suma = suma + a[i][j];
                }
            }
        }
      System.out.print("La suma de la diagonal es: "+suma);
      
    }
}