added script to split train set into small samples