pub fn diag(n: usize) -> Matrix