Integrate Gradient datasets into notebooks