Tiny tweaks to the UI