Clean up + placed code in the Yolo object