Найти сумму квадратов элементов, стоящих на диагонали матрицы

uses crt;
const n = 3;
var a : array[1..n,1..n] of integer;
    i,j : byte;
    mult : longint;
begin
     randomize();
     for i:=1 to n do
         for j:=1 to n do
             a[i,j] := random(20);
     for i:=1 to n do begin
         for j:=1 to n do
             write(a[i,j]:2,' ');
         writeln;
     end;
     mult := 1;
     for i:=1 to n do begin
         for j:=1 to n do begin
             if (i=j) then begin
                mult := mult * a[i,j];
             end;
         end;
     end;
     writeln('Result: ',mult);
     readln;
end.

uses crt;
const nmax =20;
var a : array[1..nmax,1..nmax] of integer;
    n,i,j : byte;
    sum : integer;
begin
randomize;
repeat
write('Размер матрицы до ',nmax,' n=');
read(n);
until n in [1..nmax];
for i:=1 to n do
 begin
  for j:=1 to n do
   begin
    a[i,j] := random(20);
    write(a[i,j]:3);
   end;
  writeln;
 end;
sum:=0;
for i:=1 to n do
sum:=sum+sqr(a[i,i]);
writeln('Сумма квадратов на главной диагонали=',sum);
end.