Add some hooks and UI tweaks.