Update cluster.py

a bit of cleanup