Warden::Manager.after_authentication do |user, auth, options|
  
  if Devise.respond_to?(:token_authentication_key) && auth.env["action_dispatch.request.parameters"].keys.include?(Devise.token_authentication_key.to_s)
    auth.session(options[:scope])[:login_challenge_questions] = false
    auth.session(options[:scope])[:set_challenge_questions] = false
  else
  
    if user.respond_to?(:login_challenge_questions?)
      auth.session(options[:scope])[:login_challenge_questions] = user.login_challenge_questions?(auth.request)
    end
  
    if user.respond_to?(:set_challenge_questions?)
      auth.session(options[:scope])[:set_challenge_questions] = user.set_challenge_questions?(auth.request)
    end
    
  end

end