Add ActiveRecord associations for models