Added batch size in script arguments.