Updated train.rb and evaluate.rb