minor improvements
UI