Update clustering.py