Add a size parameter to generators