Updating some yolo stuff