adding styles + improvement ui