Added feature for trainingprogram