Add flash after clicking edit