changed batch size to 8 from 64