Update application_controller to match docs