Working towards filling in more of the python wrappers for DistMatrix