Fixed KMeans.
It's working now, somewhat. It performs really poorly. I think it just doesn't do as well as a neural network.

Signed-off-by: loganperreault <perreault.logan@gmail.com>