Create k-means.py