Improvements in training phase