diff --git a/init.sql b/init.sql
index cd8394e..f6bc7b9 100644
--- a/init.sql
+++ b/init.sql
@@ -24,10 +24,12 @@ create index if not exists idx_training_data_geog on training_data using GIST(ge
 
 create or replace view training_duration as
 (select td_start.id as id,
+	td_start.geog as start_location,
+	td_end.geog as end_location,
 	td_start.t as start_time,
 	td_end.t as end_time,
 	date_trunc('second', (td_end.t - td_start.t)) as duration
-from (select td_row_n.t, td_row_n.id
+from (select td_row_n.t, td_row_n.id, td_row_n.geog
 		from (select *,
 				row_number() over (partition by td.training_id order by td.t asc) as row_n
 			from training_data td join training t on (t.id = td.training_id)
@@ -37,7 +39,7 @@ from (select td_row_n.t, td_row_n.id
 	
 	join
 	
-	(select td_row_n.t, td_row_n.id
+	(select td_row_n.t, td_row_n.id, td_row_n.geog
 		from (select *,
 				row_number() over (partition by td.training_id order by td.t desc) as row_n
 			from training_data td join training t on (t.id = td.training_id)