Cleaning up notebook; working on pre-processing steps