Added writing of trainSplit/testSplit for later validation