Adding support for update functions