input: 
  def store_and_query_user(user_data):
      """Store user data in a JSON file and query it back.
      >>> store_and_query_user({"Name": "John", "Age": 25, "City": "New York"})
      {'Name': 'John', 'Age': 25, 'City': 'New York'}
      """
      import json
      
      def store_user_data(data):
          with open('user_data.json', 'w') as f:
              json.dump(data, f)
  
      def get_user_data():
          with open('user_data.json') as f:
              return json.load(f)
  
      # store data in JSON file
      store_user_data(user_data)
  
      # get data
      data = get_user_data()
      return data
  
output:     import json
      
      def store_user_data(data):
          with open('user_data.json', 'w') as f:
              json.dump(data, f)
  
      def get_user_data():
          with open('user_data.json') as f:
              return json.load(f)
  
      # store data in JSON file
      store_user_data(user_data)
  
      # get data
      data = get_user_data()
      return data