script to split into training test sets added