training code added