added more models in Rails 4 notation