Update rebuild_coreference.py