Polish/ui

polished UI