add genie training code