Add support for Rails 4.x