Updated scripts to split out training and test sets.