Issue 2. Make the UI nicer looking.