Improved some methods for easier testing. Tried to incorporate the main K-Means logic inside a while loop, but I think I am confused