Changed matrix to identity matrix