[FIX] improve layout, styling, and ui