updated with new model and print.train info