Add support for update-script