* Fixed ngram bug in train script