Added tests and enabled n_clusters_max