Got the code working for identity matrix multiplication