Improve the JSON output