add update to generator