Update function signatures

Update function signatures for update and remove after hooks