minor bugs
ui changes