This is the script that will combine the train and test data and subset
the required data