Module: (Python) Square matrices