Updated KMeans with new code to allow us to read in training data. This
code is commented out at the moment in the constructor for KMeans