Forgotten DataLoader and NeuralNetork classes