UI fixes

css