Update for Rails 4