Update annotate_models.rb