Script for dividing data into training/test