Update train_spec.rb