batch train and test; train broken