add train.rb