Extend train_nplm with various options