Improved UX

Added background color and spiced up the UI.