Add incomplete training code