batch training is working