some little modify to the ui