More improvement in MlpBigram.py