Splitting data for Decision tree