Update klp_kmeans.py

GPU additions as comments