change the css and make UI look better