Own training code example edited.