Update train.rb