First draft of cluster.py