diff --git a/.ipynb_checkpoints/main-checkpoint.ipynb b/.ipynb_checkpoints/main-checkpoint.ipynb
index 0242be4..c6ec173 100644
--- a/.ipynb_checkpoints/main-checkpoint.ipynb
+++ b/.ipynb_checkpoints/main-checkpoint.ipynb
@@ -109,7 +109,7 @@
     "        final_scenario_df['Segment'] = 'Default'\n",
     "        final_scenario_df['SAR_FLAG'] = 'N'\n",
     "        final_scenario_df['Risk'] = 'Medium Risk'\n",
-    "        # final_scenario_df.dropna(inplace=True)\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 0242be4..c6ec173 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -109,7 +109,7 @@
     "        final_scenario_df['Segment'] = 'Default'\n",
     "        final_scenario_df['SAR_FLAG'] = 'N'\n",
     "        final_scenario_df['Risk'] = 'Medium Risk'\n",
-    "        # final_scenario_df.dropna(inplace=True)\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 93eece1..b279602 100644
--- a/main.py
+++ b/main.py
@@ -106,7 +106,7 @@ class Scenario:
         final_scenario_df['Segment'] = 'Default'
         final_scenario_df['SAR_FLAG'] = 'N'
         final_scenario_df['Risk'] = 'Medium Risk'
-        # final_scenario_df.dropna(inplace=True)
+        final_scenario_df.dropna(inplace=True)
         # final_scenario_df['RUN_DATE'] = final_scenario_df['END_DATE']
         return final_scenario_df