diff --git a/.ipynb_checkpoints/main-checkpoint.ipynb b/.ipynb_checkpoints/main-checkpoint.ipynb
index bf1f6db..31ff92b 100644
--- a/.ipynb_checkpoints/main-checkpoint.ipynb
+++ b/.ipynb_checkpoints/main-checkpoint.ipynb
@@ -95,7 +95,7 @@
     "        cols = [\n",
     "        'START_DATE_TIME',\n",
     "        'END_DATE_TIME',\n",
-    "        'FOCAL_ID',\n",
+    "        'Focal_id',\n",
     "        'MIN_PRICE',\n",
     "        'MAX_PRICE',\n",
     "        'PRICE_CHANGE_PCT',\n",
@@ -106,9 +106,9 @@
     "        final_scenario_df = pd.DataFrame(row_list, columns = cols)\n",
     "        final_scenario_df['PARTICIPANT_VOLUME_PCT'] = final_scenario_df['PARTICIPANT_VOLUME']/\\\n",
     "        final_scenario_df['TOTAL_VOLUME'] * 100\n",
-    "        final_scenario_df['SEGMENT'] = 'Default'\n",
-    "        final_scenario_df['SAR_FLAG'] = 'N'\n",
-    "        final_scenario_df['RISK'] = 'Medium Risk'\n",
+    "        final_scenario_df['Segment'] = 1\n",
+    "        final_scenario_df['SAR_FLAG'] = 1\n",
+    "        final_scenario_df['Risk'] = 1\n",
     "        final_scenario_df.dropna(inplace=True)\n",
     "        # final_scenario_df['RUN_DATE'] = final_scenario_df['END_DATE']\n",
     "        return final_scenario_df\n"
diff --git a/main.ipynb b/main.ipynb
index bf1f6db..31ff92b 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -95,7 +95,7 @@
     "        cols = [\n",
     "        'START_DATE_TIME',\n",
     "        'END_DATE_TIME',\n",
-    "        'FOCAL_ID',\n",
+    "        'Focal_id',\n",
     "        'MIN_PRICE',\n",
     "        'MAX_PRICE',\n",
     "        'PRICE_CHANGE_PCT',\n",
@@ -106,9 +106,9 @@
     "        final_scenario_df = pd.DataFrame(row_list, columns = cols)\n",
     "        final_scenario_df['PARTICIPANT_VOLUME_PCT'] = final_scenario_df['PARTICIPANT_VOLUME']/\\\n",
     "        final_scenario_df['TOTAL_VOLUME'] * 100\n",
-    "        final_scenario_df['SEGMENT'] = 'Default'\n",
-    "        final_scenario_df['SAR_FLAG'] = 'N'\n",
-    "        final_scenario_df['RISK'] = 'Medium Risk'\n",
+    "        final_scenario_df['Segment'] = 1\n",
+    "        final_scenario_df['SAR_FLAG'] = 1\n",
+    "        final_scenario_df['Risk'] = 1\n",
     "        final_scenario_df.dropna(inplace=True)\n",
     "        # final_scenario_df['RUN_DATE'] = final_scenario_df['END_DATE']\n",
     "        return final_scenario_df\n"
diff --git a/main.py b/main.py
index 401ba63..f1a001e 100644
--- a/main.py
+++ b/main.py
@@ -92,7 +92,7 @@ class Scenario:
         cols = [
         'START_DATE_TIME',
         'END_DATE_TIME',
-        'FOCAL_ID',
+        'Focal_id',
         'MIN_PRICE',
         'MAX_PRICE',
         'PRICE_CHANGE_PCT',
@@ -103,9 +103,9 @@ class Scenario:
         final_scenario_df = pd.DataFrame(row_list, columns = cols)
         final_scenario_df['PARTICIPANT_VOLUME_PCT'] = final_scenario_df['PARTICIPANT_VOLUME']/\
         final_scenario_df['TOTAL_VOLUME'] * 100
-        final_scenario_df['SEGMENT'] = 'Default'
-        final_scenario_df['SAR_FLAG'] = 'N'
-        final_scenario_df['RISK'] = 'Medium Risk'
+        final_scenario_df['Segment'] = 1
+        final_scenario_df['SAR_FLAG'] = 1
+        final_scenario_df['Risk'] = 1
         final_scenario_df.dropna(inplace=True)
         # final_scenario_df['RUN_DATE'] = final_scenario_df['END_DATE']
         return final_scenario_df