Refactor: add _format_datasets