Make the UI nicer