Added flag support for update