Added configuration parameter for batch size