changed batch size to 1000