diff --git a/.ipynb_checkpoints/main-checkpoint.ipynb b/.ipynb_checkpoints/main-checkpoint.ipynb
index dd6bb7f..46ea052 100644
--- a/.ipynb_checkpoints/main-checkpoint.ipynb
+++ b/.ipynb_checkpoints/main-checkpoint.ipynb
@@ -152,7 +152,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 20,
    "id": "82c2152f-513c-4fde-a4a9-6ee3a01ef897",
    "metadata": {
     "tags": []
@@ -202,9 +202,9 @@
     "        df_list.append(df_party)\n",
     "\n",
     "    final_df = pd.concat(df_list, ignore_index=True) \n",
-    "    Segment = final_df.groupby('Focal_id')['Segment'].agg('max').reset_index()\n",
-    "    Risk = final_df.groupby('Focal_id')['Risk'].agg('max').reset_index()\n",
-    "    SAR_FLAG = final_df.groupby('Focal_id')['SAR_FLAG'].agg('max').reset_index()\n",
+    "    Segment = data1.groupby('Focal_id')['Segment'].agg('max').reset_index()\n",
+    "    Risk = data1.groupby('Focal_id')['Risk'].agg('max').reset_index()\n",
+    "    SAR_FLAG = data1.groupby('Focal_id')['SAR_FLAG'].agg('max').reset_index()\n",
     "    \n",
     "    final_df = final_df.merge(Segment,on = 'Focal_id', how = 'left')\n",
     "    final_df = final_df.merge(Risk,on = 'Focal_id', how = 'left')\n",
diff --git a/main.ipynb b/main.ipynb
index dd6bb7f..46ea052 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -152,7 +152,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 20,
    "id": "82c2152f-513c-4fde-a4a9-6ee3a01ef897",
    "metadata": {
     "tags": []
@@ -202,9 +202,9 @@
     "        df_list.append(df_party)\n",
     "\n",
     "    final_df = pd.concat(df_list, ignore_index=True) \n",
-    "    Segment = final_df.groupby('Focal_id')['Segment'].agg('max').reset_index()\n",
-    "    Risk = final_df.groupby('Focal_id')['Risk'].agg('max').reset_index()\n",
-    "    SAR_FLAG = final_df.groupby('Focal_id')['SAR_FLAG'].agg('max').reset_index()\n",
+    "    Segment = data1.groupby('Focal_id')['Segment'].agg('max').reset_index()\n",
+    "    Risk = data1.groupby('Focal_id')['Risk'].agg('max').reset_index()\n",
+    "    SAR_FLAG = data1.groupby('Focal_id')['SAR_FLAG'].agg('max').reset_index()\n",
     "    \n",
     "    final_df = final_df.merge(Segment,on = 'Focal_id', how = 'left')\n",
     "    final_df = final_df.merge(Risk,on = 'Focal_id', how = 'left')\n",
diff --git a/main.py b/main.py
index 1cc9575..b9f9b92 100644
--- a/main.py
+++ b/main.py
@@ -139,7 +139,7 @@ query = """
 """
 
 
-# In[16]:
+# In[20]:
 
 
 def trx_count_sum_groupwise(data_filt_partywise): 
@@ -185,9 +185,9 @@ def scenario9_data(data1):
         df_list.append(df_party)
 
     final_df = pd.concat(df_list, ignore_index=True) 
-    Segment = final_df.groupby('Focal_id')['Segment'].agg('max').reset_index()
-    Risk = final_df.groupby('Focal_id')['Risk'].agg('max').reset_index()
-    SAR_FLAG = final_df.groupby('Focal_id')['SAR_FLAG'].agg('max').reset_index()
+    Segment = data1.groupby('Focal_id')['Segment'].agg('max').reset_index()
+    Risk = data1.groupby('Focal_id')['Risk'].agg('max').reset_index()
+    SAR_FLAG = data1.groupby('Focal_id')['SAR_FLAG'].agg('max').reset_index()
     
     final_df = final_df.merge(Segment,on = 'Focal_id', how = 'left')
     final_df = final_df.merge(Risk,on = 'Focal_id', how = 'left')