Finished the code assignment for the K-Means clustering.