Added training data loader and unit tests.