Removed setting batch size