Added train model call.