added an identity matrix function