added scripts for training and testing