Made the load data code into a class
Turned the training set stats into a ipython notebook (and added stats on test and validation sets)
Old changes to the old data set stats file