Creating a dataset and loaders