Makes batch size customisable