diff --git a/.ipynb_checkpoints/main-checkpoint.ipynb b/.ipynb_checkpoints/main-checkpoint.ipynb
index 1ff2132..423142d 100644
--- a/.ipynb_checkpoints/main-checkpoint.ipynb
+++ b/.ipynb_checkpoints/main-checkpoint.ipynb
@@ -94,12 +94,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 1,
    "id": "b6c85de2-6a47-4109-8885-c138c289ec25",
    "metadata": {
     "tags": []
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "ename": "IndentationError",
+     "evalue": "unexpected indent (2236745286.py, line 63)",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;36m  Cell \u001b[0;32mIn[1], line 63\u001b[0;36m\u001b[0m\n\u001b[0;31m    \"Segment\", \"Risk\", \"SAR_FLAG\"\u001b[0m\n\u001b[0m    ^\u001b[0m\n\u001b[0;31mIndentationError\u001b[0m\u001b[0;31m:\u001b[0m unexpected indent\n"
+     ]
+    }
+   ],
    "source": [
     "import pandas as pd\n",
     "\n",
@@ -162,8 +171,6 @@
     "                                                    acc_data=\"account_data_v1\",\n",
     "                                                    alert_data=\"alert_data_v1\")\n",
     "                                       )\n",
-    "        \"Focal_id\", \"Cash_deposit_total\", \"Cash_deposit_count\",\n",
-    "                \"Segment\", \"Risk\", \"SAR_FLAG\"\n",
     "        cols = [\n",
     "                \"transaction_key\",\n",
     "                \"transaction_date\",\n",
diff --git a/main.ipynb b/main.ipynb
index 1ff2132..423142d 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -94,12 +94,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 1,
    "id": "b6c85de2-6a47-4109-8885-c138c289ec25",
    "metadata": {
     "tags": []
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "ename": "IndentationError",
+     "evalue": "unexpected indent (2236745286.py, line 63)",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;36m  Cell \u001b[0;32mIn[1], line 63\u001b[0;36m\u001b[0m\n\u001b[0;31m    \"Segment\", \"Risk\", \"SAR_FLAG\"\u001b[0m\n\u001b[0m    ^\u001b[0m\n\u001b[0;31mIndentationError\u001b[0m\u001b[0;31m:\u001b[0m unexpected indent\n"
+     ]
+    }
+   ],
    "source": [
     "import pandas as pd\n",
     "\n",
@@ -162,8 +171,6 @@
     "                                                    acc_data=\"account_data_v1\",\n",
     "                                                    alert_data=\"alert_data_v1\")\n",
     "                                       )\n",
-    "        \"Focal_id\", \"Cash_deposit_total\", \"Cash_deposit_count\",\n",
-    "                \"Segment\", \"Risk\", \"SAR_FLAG\"\n",
     "        cols = [\n",
     "                \"transaction_key\",\n",
     "                \"transaction_date\",\n",
diff --git a/main.py b/main.py
index 6f052eb..557da8d 100644
--- a/main.py
+++ b/main.py
@@ -87,7 +87,7 @@
 #         return df
 
 
-# In[2]:
+# In[1]:
 
 
 import pandas as pd
@@ -151,8 +151,6 @@ class Scenario:
                                                     acc_data="account_data_v1",
                                                     alert_data="alert_data_v1")
                                        )
-        "Focal_id", "Cash_deposit_total", "Cash_deposit_count",
-                "Segment", "Risk", "SAR_FLAG"
         cols = [
                 "transaction_key",
                 "transaction_date",