I was thinking that Okani could take over his training, he's still a Genin and she is the only Jounin from the hidden Rain.