new constructor in train model