working on clustering.py