bug fixes, UI enhancements