Update Training.py