get_cluster added
add an option to plot clusters