Add Rails 4 support