Update getCluster.py