Changed get_features into get_dataset_features