Update ClusteringWithGaussians.py