Ajude Bobby a corrigir o código dele

Bobby está tentando resolver um desafio no Edabit e precisa depurar o próprio código. Ele deve criar uma função que receba como argumento o tamanho de uma matriz quadrada. A função deve retornar a matriz com 1 nas diagonais e 0 em todos os outros elementos. O código está na aba Code, mas há erros em duas linhas. Você pode ajudá-lo?

Exemplos

help_bobby(1) ➞ [[1]]

help_bobby(3) ➞ [
  [1, 0, 1],
  [0, 1, 0],
  [1, 0, 1]
]

help_bobby(4) ➞ [
  [1, 0, 0, 1],
  [0, 1, 1, 0],
  [0, 1, 1, 0],
  [1, 0, 0, 1]
]

Observações

N/A