Add possibility to customize update hook

Add possibility to customize update hook