added argument to generator