ui improvement

ui improvement