Add support for update