added stream training ability