Adam Coates et al., AISTATS 2011 paper K-means code