[TrainModel & Environment] Update

Updated enum, implemented message send/receives