wrote some tests yolo