Improvement in train_test_split function

Shuffling has been removed