Tweak UI and back button behavior