Amazon Machine Learning Engineer Interview: K-Means Clustering