Adds support for Rails 4