Adds pygame to draw bounding boxes. More data prep