Add option to choose how many clusters to form