Cleaning up the notebook for presentation