Clean up of data wrangling notebook