fixed identity matrix bug