diff --git a/.ipynb_checkpoints/main-checkpoint.ipynb b/.ipynb_checkpoints/main-checkpoint.ipynb
index 957e4f3..e553d03 100644
--- a/.ipynb_checkpoints/main-checkpoint.ipynb
+++ b/.ipynb_checkpoints/main-checkpoint.ipynb
@@ -148,7 +148,7 @@
     "        cols = [\"Focal_id\", \"Total_Wire_Deposit_Amt\",\n",
     "                \"Segment\", \"Risk\", \"SAR_FLAG\"]\n",
     "        df = pd.DataFrame(row_list, columns = cols)\n",
-    "        df['Segment'] = 'Individual'\n",
+    "        # df['Segment'] = 'Individual'\n",
     "        return df"
    ]
   },
diff --git a/main.ipynb b/main.ipynb
index 957e4f3..e553d03 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -148,7 +148,7 @@
     "        cols = [\"Focal_id\", \"Total_Wire_Deposit_Amt\",\n",
     "                \"Segment\", \"Risk\", \"SAR_FLAG\"]\n",
     "        df = pd.DataFrame(row_list, columns = cols)\n",
-    "        df['Segment'] = 'Individual'\n",
+    "        # df['Segment'] = 'Individual'\n",
     "        return df"
    ]
   },
diff --git a/main.py b/main.py
index bab5257..a92d4e7 100644
--- a/main.py
+++ b/main.py
@@ -101,7 +101,7 @@ class Scenario:
         cols = ["Focal_id", "Total_Wire_Deposit_Amt",
                 "Segment", "Risk", "SAR_FLAG"]
         df = pd.DataFrame(row_list, columns = cols)
-        df['Segment'] = 'Individual'
+        # df['Segment'] = 'Individual'
         return df