fixed the identity matrix
fixed the identity matrix