Improved JSON wrapping