Adding updated training scripts to repo