small enhancements in ui