Update kmeans.py

'k' is not a proper argument for KMeans. The name is "n_clusters"