Added notebook to prototype the pretraining