Added a training.py module that will handle all training operations