minor cleanup in ui