Put layout logical into LayoutBuilder
Change UI
Fix bugs in medium layout