added the ability to set the dataset name in the completion modal as well