Refactor JSON module