soma dos elementos da diagonal pricipal usando laço

#include <stdio.h>

int main(){

    int i, j, soma = 0, mat[3][3];

    printf("Entre com quatro elementos: \n");

    for(i = 0; i < 3; i++){
        for(j = 0; j < 3; j++){
            scanf("%d",&mat[i][j]);
        }
    }

    for(i = 0; i < 3; i++){
            soma = soma + mat[i][i];
    }

    printf("A soma: %d", soma);
}