diff --git a/.ipynb_checkpoints/main-checkpoint.ipynb b/.ipynb_checkpoints/main-checkpoint.ipynb
index c6ec173..fc80075 100644
--- a/.ipynb_checkpoints/main-checkpoint.ipynb
+++ b/.ipynb_checkpoints/main-checkpoint.ipynb
@@ -108,7 +108,7 @@
     "        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['Risk'] = 'Medium Risk'\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 c6ec173..fc80075 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -108,7 +108,7 @@
     "        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['Risk'] = 'Medium Risk'\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 b279602..e5e54a8 100644
--- a/main.py
+++ b/main.py
@@ -105,7 +105,7 @@ class Scenario:
         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['Risk'] = 'Medium Risk'
         final_scenario_df.dropna(inplace=True)
         # final_scenario_df['RUN_DATE'] = final_scenario_df['END_DATE']
         return final_scenario_df