Update k_means.py