added model trainer
added model trainer