diff --git a/Analyze.py b/Analyze.py
index 3f07dbf..1512d02 100644
--- a/Analyze.py
+++ b/Analyze.py
@@ -10,11 +10,11 @@ def import_data():
     return pd.read_csv('ttc-bus-delay-data-2023.csv')
 
 
-def compute_avg_delay(data):
+def compute_avg_delay(data, group_by: str = 'Route'):
     """
-    Compute the average delay for each route
+    Compute the average delay aggregated by specified column
     """
-    return data.groupby('Route').mean().reset_index()
+    return data.groupby(group_by).mean().reset_index()
 
 
 def main():
@@ -22,7 +22,7 @@ def main():
     Main function
     """
     data = import_data()
-    avg_delay = compute_avg_delay(data)
+    avg_delay = compute_avg_delay(data, group_by='Route')
 
     plt.hist(avg_delay['Min Delay'], bins=30)
     plt.title('Average Delay Distribution')