added gradient because yolo