adding split dataset code