public JSONObject getAsJSONObject(OntologyManager ontMgr) throws JSONException {
    JSONObject obj = new JSONObject();
    JSONArray arr = new JSONArray();

    // Need to sort
    HashMap<String, Double> sortedMap = Util.sortHashMap(scoreMap);

    for (String label : sortedMap.keySet()) {
      JSONObject oj = new JSONObject();

      // Check if the type contains domain
      if (label.contains("|")) {
        URI domainURI = ontMgr.getURIFromString(label.split("\\|")[0]);
        URI typeURI = ontMgr.getURIFromString(label.split("\\|")[1]);
        if (domainURI == null || typeURI == null) continue;
        oj.put(
            SemanticTypesUpdate.JsonKeys.DisplayDomainLabel.name(),
            domainURI.getLocalNameWithPrefixIfAvailable());
        oj.put(SemanticTypesUpdate.JsonKeys.Domain.name(), label.split("\\|")[0]);
        oj.put(
            SemanticTypesUpdate.JsonKeys.DisplayLabel.name(),
            typeURI.getLocalNameWithPrefixIfAvailable());
        oj.put(SemanticTypesUpdate.JsonKeys.FullType.name(), label.split("\\|")[1]);
      } else {
        URI typeURI = ontMgr.getURIFromString(label);
        if (typeURI == null) continue;
        oj.put(SemanticTypesUpdate.JsonKeys.FullType.name(), label);
        oj.put(
            SemanticTypesUpdate.JsonKeys.DisplayLabel.name(),
            typeURI.getLocalNameWithPrefixIfAvailable());
        oj.put(SemanticTypesUpdate.JsonKeys.DisplayDomainLabel.name(), "");
        oj.put(SemanticTypesUpdate.JsonKeys.Domain.name(), "");
      }

      oj.put("Probability", scoreMap.get(label));
      arr.put(oj);
    }
    obj.put("Labels", arr);
    return obj;
  }