Add train.rb