import ulab
print(ulab.eye(3))