Add tests for JSON structure