fixed training feature bugs