Update MCDistMat.py