Add support for Rails 4