Added feature to add and remove data sets