initial attempt at Kmeans in python