adding test to yolo