UI improvements
add
ui fixes
GUI