Some yolo updates