Update to support Rails 4