Clean up notebook utils