Update annotate_models