Changed examples to be more consistent with each other, using train_test_split