need to change train_vals to raz model
I think we're good now
xonflict merge done good