Loading and classifying the dataset