Added feature to update data sets