K-Means Clustering from Scratch - Machine Learning Python