train 2 models instead