Create kmeans.py