reduced redundancy using function for plot
Merge pull request #1863 from mazumdarparijat/KMeans_notebook

minor change in KMeans notebook