6. Identity Matrix

Easy

Return the n×n identity matrix: 1s on the main diagonal, 0s elsewhere.

Implement solve(...)