Revert "Update cluster.py"

This reverts commit e1215837a8954ccc803058973a5098e9e9af3faf.