Created train.rb file