UI changes and Segue fix