Creating labels for training in the next step