add scripts for training