Decided on kmeans function but added code for kmeans2