Update cluster_analysis.py