Add method for trip duration in train.rb

I had a great idea, I added a trip_duration method to train.rb