Identity matrix typo

Identity matrix correction