allow both train and test as command line input
Hacky method to add labels to the training file
Put labels in the correct order
updated xtrain_for_kevin that includes labels