Custom feature train and test arff files.

It now makes arff files for both train and test