tiny fix in train script & refactoring in nlp features.