adding option to plot only selected samples in 2d
adding option to plot only selected samples in 2d