Implemented K-Means and visualization. Code is still very messy though.