Added test and train split function