Added peak choosing so the cluster center corresponds to the value in the cluster with the highest probability