Составить программу, которая обнуляет отрицательные элементы, расположенные на главной диагонали

uses crt;
 
const
  nmax = 10;
 
type
  Matrix = array [1..nmax, 1..nmax] of real;
  
procedure EnterMatrix (var B : Matrix; var n, m : integer);
var
  i, j : integer;
begin
  Repeat
    Write('Количество строк (не больше ', nmax, ') = ');
    Readln(n);
  until n in [1..nmax];
  Repeat
    Write('Количество столбцов (не больше ', nmax, ') = ');
    Readln(m);
  until m in [1..nmax];
  Writeln('Введите элементы матрицы: ');
  for i := 1 to n do
    for j := 1 to m do  Readln(B[i, j]);
end;
 
procedure PrintMatrix (B : Matrix; n, m : integer);
var
  i, j : integer;
begin
  for i := 1 to n do
    begin
      for j := 1 to m do
        Write(B[i, j]:0:4, ' ');
      Writeln;
    end;
end;
 
procedure EditMatrix (var B : Matrix; var count : integer; n, m : integer);
var
  i  : integer;
begin
  count := 0;
  for i := 1 to n do
    if B[i, i] < 0 then 
      begin
        B[i, i] := 0;
        inc(count);
      end;
end;
 
var
  B : Matrix;
  n, m, count : integer;
 
begin
  ClrScr;
  EnterMatrix(B, n, m);
  ClrScr;
  Writeln('Исходная матрица: ');
  PrintMatrix(B, n, m);
  Writeln('Новая матрица: ');
  EditMatrix(B, count, n, m);
  PrintMatrix(B, n, m);
  Writeln('Отрицательных элементов на главной диагонали: ', count);
end.