fixed indentation in train.rb