Add support for custom datasets.