adding test and train files for reference