Made KMeansModel into a Trait.