Improve function to read dataset to divide into training and testing

[briannaburton]

In order to create a classifier we will need to have our dataset divided into training and testing, it can be easily done when you read the images.