minor tweaks to training scripts