Minor refactorings following the last commit for KMeans

Refactored the common function into a Functions util class. 
yep, fixed it