Remove file doc string
Add back in data shuffling for training