implemented kmeans++ in warm_up_plus.py