FIX: make K-Means accept 1D vectors