found bugs in trainStep, resulting fn calls add more nodes