testing this out on training code