Updated due to changes of NGram class