UI Fixes

UI Fixes