Add json test logic