main: refactor to abide to submission guidelines (train/prediction)
make: refactor to abide to submission guidelines (train/prediction)
doc: update for new CLI