Add things to do before a training - fixed #25