Updated UI

Changed the UI for new functions to be added.