```ts
export function cellAt(grid: number[][], row: number, column: number): number | undefined {
  if (grid[row] === undefined || grid[row][column] === undefined) return undefined;
  return grid[row][column];
}

export function diagOf(grid: number[][]): number[] {
  const result: number[] = [];
  for (let i = 0; i < grid.length; i++) {
    result.push(grid[i][i]);
  }
  return result;
}
```