Implement `diag`
Add tests for `diag`
Implement `ones` function and `zeros` function
Add tests for `ones` and `zeros`
Refactor `eye` to use `diag`