add trains.rb file