Working submission to kaggle. Output file is written as csv, header needs to be added. Input two files needed, train.csv and test.csv