Filling in more of the python wrapper for DistMatrix