Rewrite the code; Bug fixit; Add Bigram;