Slight reorganization of DistMatrix class