Add train.rb with class Train