Cluster.py changed to incorporate probabilistic method