Simplifying argument passing for DistMatrix.py and Matrix.py