Add hooks for custom UI elements to display the update prompt.