Add split argument, data shuffling, and better output.