Updated JSON wrapper