adding new train
added the write_bigrams func