Implementing our first k-means example