Added test for matrix identity function