basic kmeans and kmeans++  python implementation