Added code to training-day