Added a few things to the Training file