https://mixrolikus.cc/video/so91bzOcaVw/creating-an-identity-matrix-in-numpy