Update dataloader.cpp