[wip] Correct ngram bug + add normalisation and mean