prettied up training script, less repetetive and more general now
minor fix