UI changes and add in menu button