diff --git a/.ipynb_checkpoints/main-checkpoint.ipynb b/.ipynb_checkpoints/main-checkpoint.ipynb
index 3a54290..21cba9a 100644
--- a/.ipynb_checkpoints/main-checkpoint.ipynb
+++ b/.ipynb_checkpoints/main-checkpoint.ipynb
@@ -176,6 +176,7 @@
     "                \"SAR_FLAG\"\n",
     "            ]\n",
     "        df = pd.DataFrame(row_list, columns = cols)\n",
+    "        df['Segment'] = 'SME'\n",
     "        return df"
    ]
   },
diff --git a/main.ipynb b/main.ipynb
index 3a54290..21cba9a 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -176,6 +176,7 @@
     "                \"SAR_FLAG\"\n",
     "            ]\n",
     "        df = pd.DataFrame(row_list, columns = cols)\n",
+    "        df['Segment'] = 'SME'\n",
     "        return df"
    ]
   },
diff --git a/main.py b/main.py
index 3c8da22..dc92076 100644
--- a/main.py
+++ b/main.py
@@ -165,6 +165,7 @@ class Scenario:
                 "SAR_FLAG"
             ]
         df = pd.DataFrame(row_list, columns = cols)
+        df['Segment'] = 'SME'
         return df