Update clean_cluster.py