Improve the MlpBigram