Added unique id message passing when training