Modified to work with Rails 4 and above.