add kMeans.py to do clustering