add batch size options