Added DataLoader test