added identity matrix code