Added training data split script