Added files to seperate training part and testing part. Also code to save features added