Add option to return JSON data