Add the full training notebook

Add the training and test data