Reduce batch size to 200