Update week1clustering.py