changed train/test data to automatic split