diff --git a/Analyze.py b/Analyze.py
index 3f07dbf..1aca83f 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('Route')['Min Delay'].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')
diff --git a/Homework_Task1.png b/Homework_Task1.png
new file mode 100644
index 0000000..8694877
Binary files /dev/null and b/Homework_Task1.png differ
diff --git a/Homework_Task2.png b/Homework_Task2.png
new file mode 100644
index 0000000..946e24c
Binary files /dev/null and b/Homework_Task2.png differ
diff --git a/Homework_Task3.png b/Homework_Task3.png
new file mode 100644
index 0000000..7190ea8
Binary files /dev/null and b/Homework_Task3.png differ