option to specify train/test files