Added flag for number of clusters and also mode