adapt scripts for new dataset