minor change in KMeans notebook

-created a plot function to reduce redundancy in KMeans notebook