update to set_track_weights model method