[TrainModel & Enviroment] Updated message enum

Started handling messages in TrainContainer