Add train.rb and initialize with model & speed