Add code to extract ground truth