Update cluster.py