Minor CSS changes and UI fixes