added more example to train (set-up the train-dataset handling)