Lesson 2 includes identity matrix