full models trained, refactor segment to be more general, cleanup