Loading and preparing the text8 dataset