Changed result of k-means cluster

Result of k-means is now clusters + classifier function.
Cleaned up the script a bit.