Tweaked variables on Train script.