A few small changes
Some improvements to n-gram