Edited the UI