working on identity matrix