UnorganizedOutputWriterTests pass new models

* Added token ids to Mention constructor