Some UI changes, animations and handle the back button