Added train/test files as command line arguments
Added debug print statements