Added train.rb file