What is the loss function that we'll be using for the control? How will we handle multiple labels? Softmax matching?