Update model_additions.rb