Add dataset_info API