Improve JSON output