Python Implementation of K-Means Clustering