Small CSS and UI improvements