K means clustering algorithm in Python