Cleaned up documentation and code style in cluster.py