added UI elements and improved appearance