Update to match app.rb