diff --git a/sagemaker-debugger/model_specific_realtime_analysis/cnn_class_activation_maps/cnn_class_activation_maps.ipynb b/sagemaker-debugger/model_specific_realtime_analysis/cnn_class_activation_maps/cnn_class_activation_maps.ipynb
index 8497145a11..f014136dff 100644
--- a/sagemaker-debugger/model_specific_realtime_analysis/cnn_class_activation_maps/cnn_class_activation_maps.ipynb
+++ b/sagemaker-debugger/model_specific_realtime_analysis/cnn_class_activation_maps/cnn_class_activation_maps.ipynb
@@ -2,6 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
+   "id": "b541313a",
    "metadata": {
     "papermill": {
      "duration": 0.019155,
@@ -36,6 +37,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "62b5363d",
    "metadata": {
     "papermill": {
      "duration": 0.019147,
@@ -56,6 +58,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "da780af2",
    "metadata": {
     "papermill": {
      "duration": 0.019026,
@@ -96,6 +99,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "9b44f9c4",
    "metadata": {
     "papermill": {
      "duration": 0.01905,
@@ -114,6 +118,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "2d3ac6c4",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:14.920576Z",
@@ -142,6 +147,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "a1aee42d",
    "metadata": {
     "papermill": {
      "duration": 0.0192,
@@ -161,6 +167,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "a9915d5d",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:15.005339Z",
@@ -187,6 +194,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "828c18e0",
    "metadata": {
     "papermill": {
      "duration": 0.034419,
@@ -204,6 +212,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "d4145b2f",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:37.112365Z",
@@ -224,13 +233,13 @@
    "source": [
     "import zipfile\n",
     "! wget https://sid.erda.dk/public/archives/daaeac0d7ce1152aea9b61d9f1e19370/GTSRB_Final_Test_Images.zip\n",
-    "# ! unzip -q GTSRB_Final_Test_Images.zip\n",
     "with zipfile.ZipFile(\"GTSRB_Final_Test_Images.zip\", 'r') as zip_ref:\n",
     "    zip_ref.extractall(\"./\")"
    ]
   },
   {
    "cell_type": "markdown",
+   "id": "d7c8da80",
    "metadata": {
     "papermill": {
      "duration": 0.117074,
@@ -248,6 +257,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "acd62e1c",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:48.961919Z",
@@ -299,6 +309,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "a13869e4",
    "metadata": {
     "papermill": {
      "duration": 0.04502,
@@ -316,6 +327,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "6e8e412a",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:49.970786Z",
@@ -347,6 +359,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "ca85be55",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:50.067962Z",
@@ -371,6 +384,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "01236230",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:50.168978Z",
@@ -394,6 +408,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "712443e3",
    "metadata": {
     "papermill": {
      "duration": 0.046074,
@@ -416,6 +431,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "b39693a0",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:53.884184Z",
@@ -452,6 +468,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "d51d31a7",
    "metadata": {
     "papermill": {
      "duration": 0.046495,
@@ -470,6 +487,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "22fb6ca6",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:54.076979Z",
@@ -502,6 +520,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "28cfba68",
    "metadata": {
     "papermill": {
      "duration": 0.046268,
@@ -520,6 +539,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "ce8a0804",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:54.272889Z",
@@ -556,6 +576,7 @@
     "        \"batch_size_val\": 4,\n",
     "        \"learning_rate\": 0.001,\n",
     "    },\n",
+    "    volume_size=100,\n",
     "    debugger_hook_config=debugger_hook_config,\n",
     "    rules=[class_imbalance_rule],\n",
     ")"
@@ -563,6 +584,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "25d94395",
    "metadata": {
     "papermill": {
      "duration": 0.047051,
@@ -580,6 +602,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "26407899",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:54.977013Z",
@@ -606,6 +629,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "e0ac1bcf",
    "metadata": {
     "papermill": {
      "duration": 0.047711,
@@ -623,6 +647,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "86fa3c49",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:55.722556Z",
@@ -646,6 +671,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "12200fdf",
    "metadata": {
     "papermill": {
      "duration": 0.048474,
@@ -664,6 +690,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "34855663",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:55.952168Z",
@@ -688,6 +715,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "5d259ffd",
    "metadata": {
     "papermill": {
      "duration": 0.049098,
@@ -705,6 +733,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "0204cf89",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:56.157269Z",
@@ -733,6 +762,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "ad3f0a46",
    "metadata": {
     "papermill": {
      "duration": 0.049753,
@@ -750,6 +780,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "16897b21",
    "metadata": {
     "execution": {
      "iopub.execute_input": "2021-05-27T00:13:56.386072Z",
@@ -785,6 +816,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "96bb3778",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -802,6 +834,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "97e790d4",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -821,6 +854,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "817bec0c",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -838,6 +872,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "154557d2",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -861,6 +896,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "4762ddf0",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -878,6 +914,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "cc3323e5",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -904,6 +941,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "cf557725",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -921,6 +959,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "0c9084d7",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -941,6 +980,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "5e15cfdc",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -958,6 +998,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "4aec1a1d",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -978,6 +1019,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "90ac5f6f",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -995,6 +1037,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "d42ee9e7",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -1049,6 +1092,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "a44476b8",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -1066,6 +1110,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "52dd891a",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -1092,6 +1137,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "81e76e18",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -1109,6 +1155,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "5e8611b9",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -1136,6 +1183,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "6a653314",
    "metadata": {
     "papermill": {
      "duration": null,
@@ -1159,6 +1207,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "17190227",
    "metadata": {
     "papermill": {
      "duration": null,