Add option to coref loading for training, set flags for Mention alignment training