Add segment options to train and eval.. . need to test still