Calculate trip time

Add the trip_calculation method to the train model