Fixing a mistake in a C prototype in DistMatrix.h and further extending the python wrappers for DistMatrix