Added proper train model