POSSIBLE HUGE BREAKTHROUGH!!!!1!!

Same as earlier(no 2nd for loop, np.sum instead), with the built-in
KMeans function commented out (slightly faster at the end).