Update to run on Rails 4