Added a script to split in training and testing