Update DistMatrix.py