bug fixes
ui bug fixes