Put the KMeansAssignment stuff in the right place