Refactoring in simple.train.