Changed the UI

Changed the UI