update basic code for model training