K Means Clustering From Scratch Python