have train_test_split split in order code example

Example: train_test_split example

train_test_split example