HW2
Working kmeans with sklearn