Update to Rails 4 syntax