Modified driver for easier testing

- made variables for training data file paths, so that testing with
different sets of training data is easy